// Copyright 2007, Google Inc.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//     * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
//     * Neither the name of Google Inc. nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

//
// Tests using global test environments.

#include <stdlib.h>
#include <stdio.h>
#include "gtest/gtest.h"
#include "src/gtest-internal-inl.h"

namespace testing {
GTEST_DECLARE_string_(filter);
}

namespace {

enum FailureType {
    NO_FAILURE,
    NON_FATAL_FAILURE,
    FATAL_FAILURE
};

// For testing using global test environments.
class MyEnvironment : public testing::Environment
{
public:
    MyEnvironment() { Reset(); }

    // Depending on the value of failure_in_set_up_, SetUp() will
    // generate a non-fatal failure, generate a fatal failure, or
    // succeed.
    virtual void SetUp()
    {
        set_up_was_run_ = true;

        switch (failure_in_set_up_) {
        case NON_FATAL_FAILURE:
            ADD_FAILURE() << "Expected non-fatal failure in global set-up.";
            break;
        case FATAL_FAILURE:
            FAIL() << "Expected fatal failure in global set-up.";
            break;
        default:
            break;
        }
    }

    // Generates a non-fatal failure.
    virtual void TearDown()
    {
        tear_down_was_run_ = true;
        ADD_FAILURE() << "Expected non-fatal failure in global tear-down.";
    }

    // Resets the state of the environment s.t. it can be reused.
    void Reset()
    {
        failure_in_set_up_ = NO_FAILURE;
        set_up_was_run_ = false;
        tear_down_was_run_ = false;
    }

    // We call this function to set the type of failure SetUp() should
    // generate.
    void set_failure_in_set_up(FailureType type)
    {
        failure_in_set_up_ = type;
    }

    // Was SetUp() run?
    bool set_up_was_run() const { return set_up_was_run_; }

    // Was TearDown() run?
    bool tear_down_was_run() const { return tear_down_was_run_; }

private:
    FailureType failure_in_set_up_;
    bool set_up_was_run_;
    bool tear_down_was_run_;
};

// Was the TEST run?
bool test_was_run;

// The sole purpose of this TEST is to enable us to check whether it
// was run.
TEST(FooTest, Bar)
{
    test_was_run = true;
}

// Prints the message and aborts the program if condition is false.
void Check(bool condition, const char *msg)
{
    if (!condition) {
        printf("FAILED: %s\n", msg);
        testing::internal::posix::Abort();
    }
}

// Runs the tests.  Return true iff successful.
//
// The 'failure' parameter specifies the type of failure that should
// be generated by the global set-up.
int RunAllTests(MyEnvironment *env, FailureType failure)
{
    env->Reset();
    env->set_failure_in_set_up(failure);
    test_was_run = false;
    testing::internal::GetUnitTestImpl()->ClearAdHocTestResult();
    return RUN_ALL_TESTS();
}

} // namespace

int main(int argc, char **argv)
{
    testing::InitGoogleTest(&argc, argv);

    // Registers a global test environment, and verifies that the
    // registration function returns its argument.
    MyEnvironment *const env = new MyEnvironment;
    Check(testing::AddGlobalTestEnvironment(env) == env,
          "AddGlobalTestEnvironment() should return its argument.");

    // Verifies that RUN_ALL_TESTS() runs the tests when the global
    // set-up is successful.
    Check(RunAllTests(env, NO_FAILURE) != 0,
          "RUN_ALL_TESTS() should return non-zero, as the global tear-down "
          "should generate a failure.");
    Check(test_was_run,
          "The tests should run, as the global set-up should generate no "
          "failure");
    Check(env->tear_down_was_run(),
          "The global tear-down should run, as the global set-up was run.");

    // Verifies that RUN_ALL_TESTS() runs the tests when the global
    // set-up generates no fatal failure.
    Check(RunAllTests(env, NON_FATAL_FAILURE) != 0,
          "RUN_ALL_TESTS() should return non-zero, as both the global set-up "
          "and the global tear-down should generate a non-fatal failure.");
    Check(test_was_run,
          "The tests should run, as the global set-up should generate no "
          "fatal failure.");
    Check(env->tear_down_was_run(),
          "The global tear-down should run, as the global set-up was run.");

    // Verifies that RUN_ALL_TESTS() runs no test when the global set-up
    // generates a fatal failure.
    Check(RunAllTests(env, FATAL_FAILURE) != 0,
          "RUN_ALL_TESTS() should return non-zero, as the global set-up "
          "should generate a fatal failure.");
    Check(!test_was_run,
          "The tests should not run, as the global set-up should generate "
          "a fatal failure.");
    Check(env->tear_down_was_run(),
          "The global tear-down should run, as the global set-up was run.");

    // Verifies that RUN_ALL_TESTS() doesn't do global set-up or
    // tear-down when there is no test to run.
    testing::GTEST_FLAG(filter) = "-*";
    Check(RunAllTests(env, NO_FAILURE) == 0,
          "RUN_ALL_TESTS() should return zero, as there is no test to run.");
    Check(!env->set_up_was_run(),
          "The global set-up should not run, as there is no test to run.");
    Check(!env->tear_down_was_run(),
          "The global tear-down should not run, "
          "as the global set-up was not run.");

    printf("PASS\n");
    return 0;
}
